iT邦幫忙

2021 iThome 鐵人賽

DAY 8
0
自我挑戰組

IOS、Python自學心得30天系列 第 8

IOS、Python自學心得30天 Day-8 tensorflow.python.keras.optimizer_v2.adam改版問題

  • 分享至 

  • xImage
  •  

前言:
經過這兩天的測試,總算知道問題出在哪裡,之前訓練模型的準確度低,是因為他根本沒訓練到,查了很久後發現,是optimizer的版本有換過,原來舊的方法會跳出警告,要用新的版本來運作,可能是在安裝 Tensorflow 的時候沒注意到版本問題,以至於程式碼無法在新的 Tensorflow 上順利運作

修改後的部分程式碼:

import tensorflow as tf
from tensorflow.python.keras import backend as K
from tensorflow.python.keras.models import Model
from tensorflow.python.keras.layers import Flatten, Dense, Dropout
from tensorflow.python.keras.applications.resnet import ResNet50
from tensorflow.python.keras.optimizer_v2.adam import Adam
from tensorflow.python.keras.preprocessing.image import ImageDataGenerator
# 使用 Adam optimizer,以較低的 learning rate 進行 fine-tuning
net_final.compile(optimizer=Adam(lr=1e-5),
                  loss='categorical_crossentropy', metrics=['accuracy'])

# 輸出整個網路結構
print(net_final.summary())

# 訓練模型
net_final.fit(train_batches,
              steps_per_epoch = train_batches.samples // BATCH_SIZE,
              validation_data = valid_batches,
              validation_steps = valid_batches.samples // BATCH_SIZE,
              epochs = NUM_EPOCHS)

上一篇
IOS、Python自學心得30天 Day-7 TensorFlow 結果預測
下一篇
IOS、Python自學心得30天 Day-9 模組訓練改善-1
系列文
IOS、Python自學心得30天30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言